if(!require("EBImage")){
install.packages("BiocManager")
BiocManager::install("EBImage")
}
if(!require("R.matlab")){
install.packages("R.matlab")
}
if(!require("readxl")){
install.packages("readxl")
}
if(!require("dplyr")){
install.packages("dplyr")
}
if(!require("readxl")){
install.packages("readxl")
}
if(!require("ggplot2")){
install.packages("ggplot2")
}
if(!require("caret")){
install.packages("caret")
}
if(!require("glmnet")){
install.packages("glmnet")
}
if(!require("WeightedROC")){
install.packages("WeightedROC")
}
if(!require("MASS")){
install.packages("MASS")
}
install.packages('ROSE')
trying URL 'https://cran.rstudio.com/bin/macosx/contrib/4.0/ROSE_0.0-3.tgz'
Content type 'application/x-gzip' length 111353 bytes (108 KB)
==================================================
downloaded 108 KB
The downloaded binary packages are in
/var/folders/h6/g930s4b173zglxv0sl72b5zr0000gn/T//RtmpYsmB72/downloaded_packages
library(R.matlab)
library(readxl)
library(dplyr)
library(EBImage)
library(ggplot2)
library(caret)
library(glmnet)
library(WeightedROC)
library(MASS)
library(ROSE)
Step 0 set work directories
set.seed(2020)
setwd('/Users/zy/Documents/GitHub/Spring2021-Project3-group-5/doc')
# here replace it with your own path or manually set it in RStudio to where this rmd file is located.
# use relative path for reproducibility
Provide directories for training images. Training images and Training fiducial points will be in different subfolders.
train_dir <- "../data/train_set/" # This will be modified for different data sets.
train_image_dir <- paste(train_dir, "images/", sep="")
train_pt_dir <- paste(train_dir, "points/", sep="")
train_label_path <- paste(train_dir, "label.csv", sep="")
Step 1: set up controls for evaluation experiments.
In this chunk, we have a set of controls for the evaluation experiments.
- (T/F) cross-validation on the training set
- (T/F) reweighting the samples for training set
- (number) K, the number of CV folds
- (T/F) process features for training set
- (T/F) run evaluation on an independent test set
- (T/F) process features for test set
run.cv <- TRUE # run cross-validation on the training set
sample.reweight <- TRUE # run sample reweighting in model training
K <- 5 # number of CV folds
run.feature.train <- TRUE # process features for training set
run.test <- TRUE # run evaluation on an independent test set
run.feature.test <- TRUE # process features for test set
Step 2: import data and train-test split
#train-test split
info <- read.csv(train_label_path)
n <- nrow(info)
n_train <- round(n*(4/5), 0)
train_idx <- sample(info$Index, n_train, replace = F)
test_idx <- setdiff(info$Index, train_idx)
If you choose to extract features from images, such as using Gabor filter, R memory will exhaust all images are read together. The solution is to repeat reading a smaller batch(e.g 100) and process them.
n_files <- length(list.files(train_image_dir))
image_list <- list()
for(i in 1:100){
image_list[[i]] <- readImage(paste0(train_image_dir, sprintf("%04d", i), ".jpg"))
}
Fiducial points are stored in matlab format. In this step, we read them and store them in a list.
#function to read fiducial points
#input: index
#output: matrix of fiducial points corresponding to the index
readMat.matrix <- function(index){
return(round(readMat(paste0(train_pt_dir, sprintf("%04d", index), ".mat"))[[1]],0))
}
#load fiducial points
fiducial_pt_list <- lapply(1:n_files, readMat.matrix)
save(fiducial_pt_list, file="../output/fiducial_pt_list.RData")
Step 3: construct features and responses
feature.R should be the wrapper for all your feature engineering functions and options. The function feature( ) should have options that correspond to different scenarios for your project and produces an R object that contains features and responses that are required by all the models you are going to evaluate later.
feature.R
- Input: list of images or fiducial point
- Output: an RData file that contains extracted features and corresponding responses
source("../lib/feature.R")
tm_feature_train <- NA
if(run.feature.train){
tm_feature_train <- system.time(dat_train <- feature(fiducial_pt_list, train_idx))
save(dat_train, file="../output/feature_train.RData")
}else{
load(file="../output/feature_train.RData")
}
tm_feature_test <- NA
if(run.feature.test){
tm_feature_test <- system.time(dat_test <- feature(fiducial_pt_list, test_idx))
save(dat_test, file="../output/feature_test.RData")
}else{
load(file="../output/feature_test.RData")
}
#rebalance the training data
balanced_train_data <- ROSE(label~.,data = dat_train)$data
save(balanced_train_data, file="../output/feature_balanced_train.RData")
Step 4: Train a classification model with training features and responses
Since there are over 6000 features, we implement the PCA method to reduce dimension according to the covariance matrix. We only retain PCs with large variance.
#separate the features from label
dat_train_new <- balanced_train_data[,-dim(balanced_train_data)[2]]
dat_test_new <- dat_test[,-dim(dat_test)[2]]
#create a vector contain target number of PCs
num.pca <- c(10,50,500,1000)
train_pca <- function(num.pca){
for(i in 1:length(num.pca)){
#start time for training the model
train.model.start = proc.time()
#run PCA
pca <- prcomp(dat_train_new)
#store for each potential PC
train_pca <- data.frame(pca$x[,1:num.pca[i]], label = balanced_train_data[dim(balanced_train_data)[2]])
pred_pca=predict(pca,dat_test_new)
test_pca=data.frame(pred_pca[,1:num.pca[i]], label = dat_test[dim(dat_test)[2]])
#fitting the lda model
lda_pca <- lda(label ~ ., data = train_pca)
#stop time for training the model
train.model.end = proc.time()
#start time for testing the model
test.model.start = proc.time()
#predict lda model
lda_pred_pca = predict(lda_pca,test_pca[-dim(test_pca)[2]])
#end time for testing the model
test.model.end = proc.time()
#test accuracy
test_accuracy=confusionMatrix(lda_pred_pca$class, test_pca$label)$overall[1]
print(list(l1=train.model.end - train.model.start,
l2=test.model.end - test.model.start,
l3=test_accuracy))
}
}
train_pca(num.pca)
$l1
user system elapsed
165.321 1.728 168.072
$l2
user system elapsed
0.002 0.000 0.002
$l3
Accuracy
0.6183333
$l1
user system elapsed
156.726 1.163 158.476
$l2
user system elapsed
0.008 0.007 0.018
$l3
Accuracy
0.6816667
$l1
user system elapsed
159.786 1.354 161.673
$l2
user system elapsed
0.027 0.004 0.031
$l3
Accuracy
0.7033333
$l1
user system elapsed
180.074 1.533 182.535
$l2
user system elapsed
0.064 0.006 0.071
$l3
Accuracy
0.7116667
By comparing the training time, test time and accuracy, we use model with 50 PCs.
pca_10 <- prcomp(dat_train_new)
train_pca_10 <- data.frame(pca_10$x[,1:50], label = balanced_train_data[dim(balanced_train_data)[2]])
pred_pca_10 <- predict(pca_10,dat_test_new)
test_pca_10 <- data.frame(pred_pca_10[,1:50], label = dat_test[dim(dat_test)[2]])
save(train_pca_10, file="../output/feature_pca_train.RData")
save(test_pca_10, file="../output/feature_pca_test.RData")
tm_train = NA
tm_train <- system.time(lda_pca_10 <- lda(label ~ ., data = train_pca_10,cv = TRUE))
Calculate the training and testing accuracy of LDA model
pred_train_lda <- predict(lda_pca_10, train_pca_10[-dim(train_pca_10)[2]])
accu_train_lda <- mean(pred_train_lda$class == train_pca_10$label)
cat("The trainig accuracy of model: LDA", "is", accu_train_lda*100, "%.\n")
The trainig accuracy of model: LDA is 71.70833 %.
tm_test = NA
if(run.test){
tm_test <- system.time(pred_test_lda <- predict(lda_pca_10, test_pca_10))
}
save(pred_test_lda, file="../output/fit_train.RData")
accu_test_lda <- mean(pred_test_lda$class == test_pca_10$label)
cat("The accuracy of model: LDA", "is", accu_test_lda*100, "%.\n")
The accuracy of model: LDA is 68.16667 %.
Summarize Running Time
Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.
cat("Time for constructing training features=", tm_feature_train[1], "s \n")
Time for constructing training features= 1.059 s
cat("Time for constructing testing features=", tm_feature_test[1], "s \n")
Time for constructing testing features= 0.198 s
cat("Time for training model=", tm_train[1], "s \n")
Time for training model= 0.061 s
cat("Time for testing model=", tm_test[1], "s \n")
Time for testing model= 0.004 s
###Reference - Du, S., Tao, Y., & Martinez, A. M. (2014). Compound facial expressions of emotion. Proceedings of the National Academy of Sciences, 111(15), E1454-E1462.
LS0tCnRpdGxlOiAicGNhICsgbGRhIG1vZGVsIgphdXRob3I6ICJaaWt1biBaaHVhbmciCm91dHB1dDoKICBwZGZfZG9jdW1lbnQ6IGRlZmF1bHQKICBodG1sX25vdGVib29rOiBkZWZhdWx0Ci0tLQoKCmBgYHtyIG1lc3NhZ2U9RkFMU0V9CmlmKCFyZXF1aXJlKCJFQkltYWdlIikpewogIGluc3RhbGwucGFja2FnZXMoIkJpb2NNYW5hZ2VyIikKICBCaW9jTWFuYWdlcjo6aW5zdGFsbCgiRUJJbWFnZSIpCn0KaWYoIXJlcXVpcmUoIlIubWF0bGFiIikpewogIGluc3RhbGwucGFja2FnZXMoIlIubWF0bGFiIikKfQppZighcmVxdWlyZSgicmVhZHhsIikpewogIGluc3RhbGwucGFja2FnZXMoInJlYWR4bCIpCn0KCmlmKCFyZXF1aXJlKCJkcGx5ciIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJkcGx5ciIpCn0KaWYoIXJlcXVpcmUoInJlYWR4bCIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJyZWFkeGwiKQp9CgppZighcmVxdWlyZSgiZ2dwbG90MiIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJnZ3Bsb3QyIikKfQoKaWYoIXJlcXVpcmUoImNhcmV0IikpewogIGluc3RhbGwucGFja2FnZXMoImNhcmV0IikKfQoKaWYoIXJlcXVpcmUoImdsbW5ldCIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJnbG1uZXQiKQp9CgppZighcmVxdWlyZSgiV2VpZ2h0ZWRST0MiKSl7CiAgaW5zdGFsbC5wYWNrYWdlcygiV2VpZ2h0ZWRST0MiKQp9CgppZighcmVxdWlyZSgiTUFTUyIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJNQVNTIikKfQoKaW5zdGFsbC5wYWNrYWdlcygnUk9TRScpCgpsaWJyYXJ5KFIubWF0bGFiKQpsaWJyYXJ5KHJlYWR4bCkKbGlicmFyeShkcGx5cikKbGlicmFyeShFQkltYWdlKQpsaWJyYXJ5KGdncGxvdDIpCmxpYnJhcnkoY2FyZXQpCmxpYnJhcnkoZ2xtbmV0KQpsaWJyYXJ5KFdlaWdodGVkUk9DKQpsaWJyYXJ5KE1BU1MpCmxpYnJhcnkoUk9TRSkKCmBgYAoKIyMjIFN0ZXAgMCBzZXQgd29yayBkaXJlY3RvcmllcwpgYGB7ciB3a2RpciwgZXZhbD1GQUxTRX0Kc2V0LnNlZWQoMjAyMCkKc2V0d2QoJy9Vc2Vycy96eS9Eb2N1bWVudHMvR2l0SHViL1NwcmluZzIwMjEtUHJvamVjdDMtZ3JvdXAtNS9kb2MnKQojIGhlcmUgcmVwbGFjZSBpdCB3aXRoIHlvdXIgb3duIHBhdGggb3IgbWFudWFsbHkgc2V0IGl0IGluIFJTdHVkaW8gdG8gd2hlcmUgdGhpcyBybWQgZmlsZSBpcyBsb2NhdGVkLiAKIyB1c2UgcmVsYXRpdmUgcGF0aCBmb3IgcmVwcm9kdWNpYmlsaXR5CmBgYAoKUHJvdmlkZSBkaXJlY3RvcmllcyBmb3IgdHJhaW5pbmcgaW1hZ2VzLiBUcmFpbmluZyBpbWFnZXMgYW5kIFRyYWluaW5nIGZpZHVjaWFsIHBvaW50cyB3aWxsIGJlIGluIGRpZmZlcmVudCBzdWJmb2xkZXJzLiAKYGBge3J9CnRyYWluX2RpciA8LSAiLi4vZGF0YS90cmFpbl9zZXQvIiAjIFRoaXMgd2lsbCBiZSBtb2RpZmllZCBmb3IgZGlmZmVyZW50IGRhdGEgc2V0cy4KdHJhaW5faW1hZ2VfZGlyIDwtIHBhc3RlKHRyYWluX2RpciwgImltYWdlcy8iLCBzZXA9IiIpCnRyYWluX3B0X2RpciA8LSBwYXN0ZSh0cmFpbl9kaXIsICAicG9pbnRzLyIsIHNlcD0iIikKdHJhaW5fbGFiZWxfcGF0aCA8LSBwYXN0ZSh0cmFpbl9kaXIsICJsYWJlbC5jc3YiLCBzZXA9IiIpIApgYGAKCiMjIyBTdGVwIDE6IHNldCB1cCBjb250cm9scyBmb3IgZXZhbHVhdGlvbiBleHBlcmltZW50cy4KCkluIHRoaXMgY2h1bmssIHdlIGhhdmUgYSBzZXQgb2YgY29udHJvbHMgZm9yIHRoZSBldmFsdWF0aW9uIGV4cGVyaW1lbnRzLiAKCisgKFQvRikgY3Jvc3MtdmFsaWRhdGlvbiBvbiB0aGUgdHJhaW5pbmcgc2V0CisgKFQvRikgcmV3ZWlnaHRpbmcgdGhlIHNhbXBsZXMgZm9yIHRyYWluaW5nIHNldCAKKyAobnVtYmVyKSBLLCB0aGUgbnVtYmVyIG9mIENWIGZvbGRzCisgKFQvRikgcHJvY2VzcyBmZWF0dXJlcyBmb3IgdHJhaW5pbmcgc2V0CisgKFQvRikgcnVuIGV2YWx1YXRpb24gb24gYW4gaW5kZXBlbmRlbnQgdGVzdCBzZXQKKyAoVC9GKSBwcm9jZXNzIGZlYXR1cmVzIGZvciB0ZXN0IHNldAoKYGBge3IgZXhwX3NldHVwfQpydW4uY3YgPC0gVFJVRSAjIHJ1biBjcm9zcy12YWxpZGF0aW9uIG9uIHRoZSB0cmFpbmluZyBzZXQKc2FtcGxlLnJld2VpZ2h0IDwtIFRSVUUgIyBydW4gc2FtcGxlIHJld2VpZ2h0aW5nIGluIG1vZGVsIHRyYWluaW5nCksgPC0gNSAgIyBudW1iZXIgb2YgQ1YgZm9sZHMKcnVuLmZlYXR1cmUudHJhaW4gPC0gVFJVRSAjIHByb2Nlc3MgZmVhdHVyZXMgZm9yIHRyYWluaW5nIHNldApydW4udGVzdCA8LSBUUlVFICMgcnVuIGV2YWx1YXRpb24gb24gYW4gaW5kZXBlbmRlbnQgdGVzdCBzZXQKcnVuLmZlYXR1cmUudGVzdCA8LSBUUlVFICMgcHJvY2VzcyBmZWF0dXJlcyBmb3IgdGVzdCBzZXQKYGBgCgoKIyMjIFN0ZXAgMjogaW1wb3J0IGRhdGEgYW5kIHRyYWluLXRlc3Qgc3BsaXQgCmBgYHtyfQojdHJhaW4tdGVzdCBzcGxpdAppbmZvIDwtIHJlYWQuY3N2KHRyYWluX2xhYmVsX3BhdGgpCm4gPC0gbnJvdyhpbmZvKQpuX3RyYWluIDwtIHJvdW5kKG4qKDQvNSksIDApCnRyYWluX2lkeCA8LSBzYW1wbGUoaW5mbyRJbmRleCwgbl90cmFpbiwgcmVwbGFjZSA9IEYpCnRlc3RfaWR4IDwtIHNldGRpZmYoaW5mbyRJbmRleCwgdHJhaW5faWR4KQoKYGBgCgpJZiB5b3UgY2hvb3NlIHRvIGV4dHJhY3QgZmVhdHVyZXMgZnJvbSBpbWFnZXMsIHN1Y2ggYXMgdXNpbmcgR2Fib3IgZmlsdGVyLCBSIG1lbW9yeSB3aWxsIGV4aGF1c3QgYWxsIGltYWdlcyBhcmUgcmVhZCB0b2dldGhlci4gVGhlIHNvbHV0aW9uIGlzIHRvIHJlcGVhdCByZWFkaW5nIGEgc21hbGxlciBiYXRjaChlLmcgMTAwKSBhbmQgcHJvY2VzcyB0aGVtLiAKYGBge3J9Cm5fZmlsZXMgPC0gbGVuZ3RoKGxpc3QuZmlsZXModHJhaW5faW1hZ2VfZGlyKSkKCmltYWdlX2xpc3QgPC0gbGlzdCgpCmZvcihpIGluIDE6MTAwKXsKICAgaW1hZ2VfbGlzdFtbaV1dIDwtIHJlYWRJbWFnZShwYXN0ZTAodHJhaW5faW1hZ2VfZGlyLCBzcHJpbnRmKCIlMDRkIiwgaSksICIuanBnIikpCn0KYGBgCgpGaWR1Y2lhbCBwb2ludHMgYXJlIHN0b3JlZCBpbiBtYXRsYWIgZm9ybWF0LiBJbiB0aGlzIHN0ZXAsIHdlIHJlYWQgdGhlbSBhbmQgc3RvcmUgdGhlbSBpbiBhIGxpc3QuCmBgYHtyIHJlYWQgZmlkdWNpYWwgcG9pbnRzfQojZnVuY3Rpb24gdG8gcmVhZCBmaWR1Y2lhbCBwb2ludHMKI2lucHV0OiBpbmRleAojb3V0cHV0OiBtYXRyaXggb2YgZmlkdWNpYWwgcG9pbnRzIGNvcnJlc3BvbmRpbmcgdG8gdGhlIGluZGV4CnJlYWRNYXQubWF0cml4IDwtIGZ1bmN0aW9uKGluZGV4KXsKICAgICByZXR1cm4ocm91bmQocmVhZE1hdChwYXN0ZTAodHJhaW5fcHRfZGlyLCBzcHJpbnRmKCIlMDRkIiwgaW5kZXgpLCAiLm1hdCIpKVtbMV1dLDApKQp9CgojbG9hZCBmaWR1Y2lhbCBwb2ludHMKZmlkdWNpYWxfcHRfbGlzdCA8LSBsYXBwbHkoMTpuX2ZpbGVzLCByZWFkTWF0Lm1hdHJpeCkKc2F2ZShmaWR1Y2lhbF9wdF9saXN0LCBmaWxlPSIuLi9vdXRwdXQvZmlkdWNpYWxfcHRfbGlzdC5SRGF0YSIpCmBgYAoKIyMjIFN0ZXAgMzogY29uc3RydWN0IGZlYXR1cmVzIGFuZCByZXNwb25zZXMKCisgVGhlIGZvbGxvdyBwbG90cyBzaG93IGhvdyBwYWlyd2lzZSBkaXN0YW5jZSBiZXR3ZWVuIGZpZHVjaWFsIHBvaW50cyBjYW4gd29yayBhcyBmZWF0dXJlIGZvciBmYWNpYWwgZW1vdGlvbiByZWNvZ25pdGlvbi4KCiAgKyBJbiB0aGUgZmlyc3QgY29sdW1uLCA3OCBmaWR1Y2lhbHMgcG9pbnRzIG9mIGVhY2ggZW1vdGlvbiBhcmUgbWFya2VkIGluIG9yZGVyLiAKICArIEluIHRoZSBzZWNvbmQgY29sdW1uIGRpc3RyaWJ1dGlvbnMgb2YgdmVydGljYWwgZGlzdGFuY2UgYmV0d2VlbiByaWdodCBwdXBpbCgxKSBhbmQgIHJpZ2h0IGJyb3cgcGVhaygyMSkgYXJlIHNob3duIGluICBoaXN0b2dyYW1zLiBGb3IgZXhhbXBsZSwgdGhlIGRpc3RhbmNlIG9mIGFuIGFuZ3J5IGZhY2UgdGVuZHMgdG8gYmUgc2hvcnRlciB0aGFuIHRoYXQgb2YgYSBzdXJwcmlzZWQgZmFjZS4KICArIFRoZSB0aGlyZCBjb2x1bW4gaXMgdGhlIGRpc3RyaWJ1dGlvbnMgb2YgdmVydGljYWwgZGlzdGFuY2VzIGJldHdlZW4gcmlnaHQgbW91dGggY29ybmVyKDUwKQphbmQgdGhlIG1pZHBvaW50IG9mIHRoZSB1cHBlciBsaXAoNTIpLiAgRm9yIGV4YW1wbGUsIHRoZSBkaXN0YW5jZSBvZiBhbiBoYXBweSBmYWNlIHRlbmRzIHRvIGJlIHNob3J0ZXIgdGhhbiB0aGF0IG9mIGEgc2FkIGZhY2UuCgohW0ZpZ3VyZTFdKC4uL2ZpZ3MvZmVhdHVyZV92aXN1YWxpemF0aW9uLmpwZykKCmBmZWF0dXJlLlJgIHNob3VsZCBiZSB0aGUgd3JhcHBlciBmb3IgYWxsIHlvdXIgZmVhdHVyZSBlbmdpbmVlcmluZyBmdW5jdGlvbnMgYW5kIG9wdGlvbnMuIFRoZSBmdW5jdGlvbiBgZmVhdHVyZSggKWAgc2hvdWxkIGhhdmUgb3B0aW9ucyB0aGF0IGNvcnJlc3BvbmQgdG8gZGlmZmVyZW50IHNjZW5hcmlvcyBmb3IgeW91ciBwcm9qZWN0IGFuZCBwcm9kdWNlcyBhbiBSIG9iamVjdCB0aGF0IGNvbnRhaW5zIGZlYXR1cmVzIGFuZCByZXNwb25zZXMgdGhhdCBhcmUgcmVxdWlyZWQgYnkgYWxsIHRoZSBtb2RlbHMgeW91IGFyZSBnb2luZyB0byBldmFsdWF0ZSBsYXRlci4gCiAgCiAgKyBgZmVhdHVyZS5SYAogICsgSW5wdXQ6IGxpc3Qgb2YgaW1hZ2VzIG9yIGZpZHVjaWFsIHBvaW50CiAgKyBPdXRwdXQ6IGFuIFJEYXRhIGZpbGUgdGhhdCBjb250YWlucyBleHRyYWN0ZWQgZmVhdHVyZXMgYW5kIGNvcnJlc3BvbmRpbmcgcmVzcG9uc2VzCgpgYGB7ciBmZWF0dXJlfQpzb3VyY2UoIi4uL2xpYi9mZWF0dXJlLlIiKQp0bV9mZWF0dXJlX3RyYWluIDwtIE5BCmlmKHJ1bi5mZWF0dXJlLnRyYWluKXsKICB0bV9mZWF0dXJlX3RyYWluIDwtIHN5c3RlbS50aW1lKGRhdF90cmFpbiA8LSBmZWF0dXJlKGZpZHVjaWFsX3B0X2xpc3QsIHRyYWluX2lkeCkpCiAgc2F2ZShkYXRfdHJhaW4sIGZpbGU9Ii4uL291dHB1dC9mZWF0dXJlX3RyYWluLlJEYXRhIikKfWVsc2V7CiAgbG9hZChmaWxlPSIuLi9vdXRwdXQvZmVhdHVyZV90cmFpbi5SRGF0YSIpCn0KCnRtX2ZlYXR1cmVfdGVzdCA8LSBOQQppZihydW4uZmVhdHVyZS50ZXN0KXsKICB0bV9mZWF0dXJlX3Rlc3QgPC0gc3lzdGVtLnRpbWUoZGF0X3Rlc3QgPC0gZmVhdHVyZShmaWR1Y2lhbF9wdF9saXN0LCB0ZXN0X2lkeCkpCiAgc2F2ZShkYXRfdGVzdCwgZmlsZT0iLi4vb3V0cHV0L2ZlYXR1cmVfdGVzdC5SRGF0YSIpCn1lbHNlewogIGxvYWQoZmlsZT0iLi4vb3V0cHV0L2ZlYXR1cmVfdGVzdC5SRGF0YSIpCn0KCiNyZWJhbGFuY2UgdGhlIHRyYWluaW5nIGRhdGEKYmFsYW5jZWRfdHJhaW5fZGF0YSA8LSBST1NFKGxhYmVsfi4sZGF0YSA9IGRhdF90cmFpbikkZGF0YQpzYXZlKGJhbGFuY2VkX3RyYWluX2RhdGEsIGZpbGU9Ii4uL291dHB1dC9mZWF0dXJlX2JhbGFuY2VkX3RyYWluLlJEYXRhIikKYGBgCgojIyMgU3RlcCA0OiBUcmFpbiBhIGNsYXNzaWZpY2F0aW9uIG1vZGVsIHdpdGggdHJhaW5pbmcgZmVhdHVyZXMgYW5kIHJlc3BvbnNlcwpTaW5jZSB0aGVyZSBhcmUgb3ZlciA2MDAwIGZlYXR1cmVzLCB3ZSBpbXBsZW1lbnQgdGhlIFBDQSBtZXRob2QgdG8gcmVkdWNlIGRpbWVuc2lvbiBhY2NvcmRpbmcgdG8gdGhlIGNvdmFyaWFuY2UgbWF0cml4LiBXZSBvbmx5IHJldGFpbiBQQ3Mgd2l0aCBsYXJnZSB2YXJpYW5jZS4KCmBgYHtyIHBjYSBsZGF9CiNzZXBhcmF0ZSB0aGUgZmVhdHVyZXMgZnJvbSBsYWJlbApkYXRfdHJhaW5fbmV3IDwtIGJhbGFuY2VkX3RyYWluX2RhdGFbLC1kaW0oYmFsYW5jZWRfdHJhaW5fZGF0YSlbMl1dCmRhdF90ZXN0X25ldyA8LSBkYXRfdGVzdFssLWRpbShkYXRfdGVzdClbMl1dCiNjcmVhdGUgYSB2ZWN0b3IgY29udGFpbiB0YXJnZXQgbnVtYmVyIG9mIFBDcwpudW0ucGNhIDwtIGMoMTAsNTAsNTAwLDEwMDApCgp0cmFpbl9wY2EgPC0gZnVuY3Rpb24obnVtLnBjYSl7CiAgZm9yKGkgaW4gMTpsZW5ndGgobnVtLnBjYSkpewogICAgI3N0YXJ0IHRpbWUgZm9yIHRyYWluaW5nIHRoZSBtb2RlbAogICAgdHJhaW4ubW9kZWwuc3RhcnQgPSBwcm9jLnRpbWUoKQogICAgI3J1biBQQ0EKICAgIHBjYSA8LSBwcmNvbXAoZGF0X3RyYWluX25ldykKICAgICNzdG9yZSBmb3IgZWFjaCBwb3RlbnRpYWwgUEMKICAgIHRyYWluX3BjYSA8LSBkYXRhLmZyYW1lKHBjYSR4WywxOm51bS5wY2FbaV1dLCBsYWJlbCA9IGJhbGFuY2VkX3RyYWluX2RhdGFbZGltKGJhbGFuY2VkX3RyYWluX2RhdGEpWzJdXSkKICAgIHByZWRfcGNhPXByZWRpY3QocGNhLGRhdF90ZXN0X25ldykKICAgIHRlc3RfcGNhPWRhdGEuZnJhbWUocHJlZF9wY2FbLDE6bnVtLnBjYVtpXV0sIGxhYmVsID0gZGF0X3Rlc3RbZGltKGRhdF90ZXN0KVsyXV0pCiAgICAjZml0dGluZyB0aGUgbGRhIG1vZGVsIAogICAgbGRhX3BjYSA8LSBsZGEobGFiZWwgfiAuLCBkYXRhID0gdHJhaW5fcGNhKSAKICAgICNzdG9wIHRpbWUgZm9yIHRyYWluaW5nIHRoZSBtb2RlbAogICAgdHJhaW4ubW9kZWwuZW5kID0gcHJvYy50aW1lKCkKICAgICNzdGFydCB0aW1lIGZvciB0ZXN0aW5nIHRoZSBtb2RlbAogICAgdGVzdC5tb2RlbC5zdGFydCA9IHByb2MudGltZSgpCiAgICAjcHJlZGljdCBsZGEgbW9kZWwKICAgIGxkYV9wcmVkX3BjYSA9IHByZWRpY3QobGRhX3BjYSx0ZXN0X3BjYVstZGltKHRlc3RfcGNhKVsyXV0pCiAgICAjZW5kIHRpbWUgZm9yIHRlc3RpbmcgdGhlIG1vZGVsCiAgICB0ZXN0Lm1vZGVsLmVuZCA9IHByb2MudGltZSgpCiAgICAjdGVzdCBhY2N1cmFjeQogICAgdGVzdF9hY2N1cmFjeT1jb25mdXNpb25NYXRyaXgobGRhX3ByZWRfcGNhJGNsYXNzLCB0ZXN0X3BjYSRsYWJlbCkkb3ZlcmFsbFsxXQogICAgcHJpbnQobGlzdChsMT10cmFpbi5tb2RlbC5lbmQgLSB0cmFpbi5tb2RlbC5zdGFydCwKICAgICAgICAgICAgICAgbDI9dGVzdC5tb2RlbC5lbmQgLSB0ZXN0Lm1vZGVsLnN0YXJ0LAogICAgICAgICAgICAgICBsMz10ZXN0X2FjY3VyYWN5KSkKICAgIH0KICB9CnRyYWluX3BjYShudW0ucGNhKQpgYGAKCkJ5IGNvbXBhcmluZyB0aGUgdHJhaW5pbmcgdGltZSwgdGVzdCB0aW1lIGFuZCBhY2N1cmFjeSwgd2UgdXNlIG1vZGVsIHdpdGggNTAgUENzLgoKYGBge3J9CnBjYV8xMCA8LSBwcmNvbXAoZGF0X3RyYWluX25ldykKdHJhaW5fcGNhXzEwIDwtIGRhdGEuZnJhbWUocGNhXzEwJHhbLDE6NTBdLCBsYWJlbCA9IGJhbGFuY2VkX3RyYWluX2RhdGFbZGltKGJhbGFuY2VkX3RyYWluX2RhdGEpWzJdXSkKcHJlZF9wY2FfMTAgPC0gcHJlZGljdChwY2FfMTAsZGF0X3Rlc3RfbmV3KQp0ZXN0X3BjYV8xMCA8LSBkYXRhLmZyYW1lKHByZWRfcGNhXzEwWywxOjUwXSwgbGFiZWwgPSBkYXRfdGVzdFtkaW0oZGF0X3Rlc3QpWzJdXSkKCnNhdmUodHJhaW5fcGNhXzEwLCBmaWxlPSIuLi9vdXRwdXQvZmVhdHVyZV9wY2FfdHJhaW4uUkRhdGEiKQpzYXZlKHRlc3RfcGNhXzEwLCBmaWxlPSIuLi9vdXRwdXQvZmVhdHVyZV9wY2FfdGVzdC5SRGF0YSIpCgp0bV90cmFpbiA9IE5BCnRtX3RyYWluIDwtIHN5c3RlbS50aW1lKGxkYV9wY2FfMTAgPC0gbGRhKGxhYmVsIH4gLiwgZGF0YSA9IHRyYWluX3BjYV8xMCxjdiA9IFRSVUUpKQoKYGBgCgpDYWxjdWxhdGUgdGhlIHRyYWluaW5nIGFuZCB0ZXN0aW5nIGFjY3VyYWN5IG9mIExEQSBtb2RlbApgYGB7cn0KcHJlZF90cmFpbl9sZGEgPC0gcHJlZGljdChsZGFfcGNhXzEwLCB0cmFpbl9wY2FfMTBbLWRpbSh0cmFpbl9wY2FfMTApWzJdXSkKYWNjdV90cmFpbl9sZGEgPC0gbWVhbihwcmVkX3RyYWluX2xkYSRjbGFzcyA9PSB0cmFpbl9wY2FfMTAkbGFiZWwpCmNhdCgiVGhlIHRyYWluaWcgYWNjdXJhY3kgb2YgbW9kZWw6IExEQSIsICJpcyIsIGFjY3VfdHJhaW5fbGRhKjEwMCwgIiUuXG4iKQoKdG1fdGVzdCA9IE5BCmlmKHJ1bi50ZXN0KXsKICB0bV90ZXN0IDwtIHN5c3RlbS50aW1lKHByZWRfdGVzdF9sZGEgPC0gcHJlZGljdChsZGFfcGNhXzEwLCB0ZXN0X3BjYV8xMCkpCn0KCnNhdmUocHJlZF90ZXN0X2xkYSwgZmlsZT0iLi4vb3V0cHV0L2ZpdF90cmFpbi5SRGF0YSIpCgphY2N1X3Rlc3RfbGRhIDwtIG1lYW4ocHJlZF90ZXN0X2xkYSRjbGFzcyA9PSB0ZXN0X3BjYV8xMCRsYWJlbCkKY2F0KCJUaGUgYWNjdXJhY3kgb2YgbW9kZWw6IExEQSIsICJpcyIsIGFjY3VfdGVzdF9sZGEqMTAwLCAiJS5cbiIpCgpgYGAKCiMjIyBTdW1tYXJpemUgUnVubmluZyBUaW1lClByZWRpY3Rpb24gcGVyZm9ybWFuY2UgbWF0dGVycywgc28gZG9lcyB0aGUgcnVubmluZyB0aW1lcyBmb3IgY29uc3RydWN0aW5nIGZlYXR1cmVzIGFuZCBmb3IgdHJhaW5pbmcgdGhlIG1vZGVsLCBlc3BlY2lhbGx5IHdoZW4gdGhlIGNvbXB1dGF0aW9uIHJlc291cmNlIGlzIGxpbWl0ZWQuIApgYGB7ciBydW5uaW5nX3RpbWV9CmNhdCgiVGltZSBmb3IgY29uc3RydWN0aW5nIHRyYWluaW5nIGZlYXR1cmVzPSIsIHRtX2ZlYXR1cmVfdHJhaW5bMV0sICJzIFxuIikKY2F0KCJUaW1lIGZvciBjb25zdHJ1Y3RpbmcgdGVzdGluZyBmZWF0dXJlcz0iLCB0bV9mZWF0dXJlX3Rlc3RbMV0sICJzIFxuIikKY2F0KCJUaW1lIGZvciB0cmFpbmluZyBtb2RlbD0iLCB0bV90cmFpblsxXSwgInMgXG4iKSAKY2F0KCJUaW1lIGZvciB0ZXN0aW5nIG1vZGVsPSIsIHRtX3Rlc3RbMV0sICJzIFxuIikKYGBgCgojIyNSZWZlcmVuY2UKLSBEdSwgUy4sIFRhbywgWS4sICYgTWFydGluZXosIEEuIE0uICgyMDE0KS4gQ29tcG91bmQgZmFjaWFsIGV4cHJlc3Npb25zIG9mIGVtb3Rpb24uIFByb2NlZWRpbmdzIG9mIHRoZSBOYXRpb25hbCBBY2FkZW15IG9mIFNjaWVuY2VzLCAxMTEoMTUpLCBFMTQ1NC1FMTQ2Mi4KCgoKCgoKCgoKCgoKCg==